package 剑指Offer.简单;

import Entity.TreeNode;

/**
 * 输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
 * <p>
 *  
 * <p>
 * 示例 1:
 * <p>
 * 给定二叉树 [3,9,20,null,null,15,7]
 * <p>
 * 3
 * / \
 * 9  20
 * /  \
 * 15   7
 * 返回 true 。
 * <p>
 * 示例 2:
 * <p>
 * 给定二叉树 [1,2,2,3,3,null,null,4,4]
 * <p>
 * 1
 * / \
 * 2   2
 * / \
 * 3   3
 * / \
 * 4   4
 * 返回 false 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/ping-heng-er-cha-shu-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author caihe
 * @date 2022/8/29 11:12
 */
public class _55II平衡二叉树 {

    /**
     * 后序遍历
     * 从底至上 不符合平衡就进行剪枝
     */
    public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        }
        return dfs(root) != -1;
    }

    int dfs(TreeNode node) {
        if (node == null) {
            return 0;
        }
        int left = dfs(node.left);
        if (left == -1){
            return left;
        }
        int right = dfs(node.right);
        if (right == -1){
            return right;
        }
        /**
         * 使用后序遍历
         *  先计算出子树的高度差 不能大于1 大于1就直接返回-1 不是平衡二叉树
         *  小于1则返回node节点的高度 （子树最大高 + 1）
         * */
        return Math.abs(left - right) < 2 ? Math.max(left, right) + 1 : -1;
    }


    /**
     * 前序遍历
     * 从顶至下 判断每个节点的平衡情况 
     */
    public boolean isBalancedV2(TreeNode root) {
        if (root == null) {
            return true;
        }
        return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalancedV2(root.left) && isBalancedV2(root.right);
    }
    
    //节点高度
    int height(TreeNode node) {
        if (node == null) {
            return 0;
        }
        return Math.max(height(node.left), height(node.right)) + 1;
    }
    
    
    

    public static void main(String[] args) {
        TreeNode t1 = new TreeNode(1);
        TreeNode t2 = new TreeNode(2);
        TreeNode t3 = new TreeNode(3);
        t1.right = t2;
        t2.right = t3;

        _55II平衡二叉树 a = new _55II平衡二叉树();
        a.isBalanced(t1);
        System.out.println();
    }

}
